﻿#region Copyright

// The contents of this file are subject to the Mozilla Public License
//  Version 1.1 (the "License"); you may not use this file except in compliance
//  with the License. You may obtain a copy of the License at
//  
//  http://www.mozilla.org/MPL/
//  
//  Software distributed under the License is distributed on an "AS IS"
//  basis, WITHOUT WARRANTY OF ANY KIND, either express or implied. See the
//  License for the specific language governing rights and limitations under 
//  the License.
//  
//  The Initial Developer of the Original Code is Robert Smyth.
//  Portions created by Robert Smyth are Copyright (C) 2008.
//  
//  All Rights Reserved.

#endregion

using System;
using NUnitGrid.Framework;
using NUnitGrid.Protocol;
using NUnitGrid.Protocol.Commands;


namespace NUnitGrid.Server.Plugins.Nodes.Node.States
{
    public class StartingState<T> : IGridNodeState, IReadInfoListener, IScheduledEventListener
        where T : IGridNode
    {
        private readonly IGridMetrics gridMetrics;
        private readonly TimeSpan infoReadDelay = TimeSpan.FromSeconds(1);
        private readonly T node;
        private readonly INodeAvailableListener<T> nodeAvailableListener;
        private readonly IGridProtocol protocol;
        private readonly IReadInfoListener readInfoListener;
        private readonly ISchedule schedule;
        private readonly IStateEngine<IGridNodeState> stateEngine;
        private readonly IServerStateFactory stateFactory;

        public StartingState(IStateEngine<IGridNodeState> stateEngine, IServerStateFactory stateFactory,
                             IGridProtocol protocol, IReadInfoListener readInfoListener, ISchedule schedule,
                             INodeAvailableListener<T> nodeAvailableListener, T node, IGridMetrics gridMetrics)
        {
            this.stateEngine = stateEngine;
            this.stateFactory = stateFactory;
            this.protocol = protocol;
            this.readInfoListener = readInfoListener;
            this.schedule = schedule;
            this.nodeAvailableListener = nodeAvailableListener;
            this.node = node;
            this.gridMetrics = gridMetrics;
        }

        public void OnEntering()
        {
            schedule.AddEvent(this, infoReadDelay);
        }

        public void Start()
        {
            throw new InvalidOperationException("Cannot start when starting.");
        }

        public void OnConnectedChanged(bool connected)
        {
            if (!connected)
            {
                schedule.RemoveEventsFor(this);
                stateEngine.ChangeState(stateFactory.CreateLinkNotConnectedState());
            }
        }

        public void RunTask(IRunOnGridTask task, IRunOnGridTaskListener taskListener,
                            IGridNodeListener<IGridNodeMetrics> nodeListener)
        {
            throw new NotImplementedException();
        }

        void IReadInfoListener.OnInfoRead(bool succeeded, string machineName, string nickName, int numberOfProcessors,
                                          int processorMHz, string softwareVersion)
        {
            if (succeeded)
            {
                readInfoListener.OnInfoRead(true, machineName, nickName, numberOfProcessors, processorMHz,
                                            softwareVersion);
                stateEngine.ChangeState(stateFactory.CreateAvailableState());
                nodeAvailableListener.OnNodeAvailable(node, NodeAvailability.Available);
            }
            else
            {
                stateEngine.ChangeState(stateFactory.CreateFailedState());
            }
        }

        void IScheduledEventListener.OnEventFired()
        {
            protocol.ReadInfo(this);
            protocol.SendMetricsUpdate(gridMetrics.BusyNodesCount, gridMetrics.OnlineNodesCount,
                                       gridMetrics.PendingTasksCount);
        }
    }
}